package com.chen;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Stream;

/**
 * 给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。
 * 请你找出并返回这两个正序数组的 中位数 。
 * 算法的时间复杂度应该为 O(log (m+n)) 。
 * 输入：nums1 = [1,3], nums2 = [2]
 * 输出：2.00000
 * 解释：合并数组 = [1,2,3] ，中位数 2
 * ====================================
 * 示例 2：
 * 输入：nums1 = [1,2], nums2 = [3,4]
 * 输出：2.50000
 * 解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5
 */
public class Test2 {
    public static void main(String[] args) {
        int[] arr1 = {1, 3};
        int[] arr2 = {2, 4, 5};
        System.out.println(test1(arr1, arr2));
    }

    public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int size = nums1.length + nums2.length;
        List<Integer> list = Stream.concat(Arrays.stream(nums1).boxed(), Arrays.stream(nums2).boxed()).sorted().toList();
        if (size % 2 == 0) {
            int index = (size / 2) - 1;
            return (list.get(index) + list.get(index + 1)) / 2.0;
        } else {
            return list.get(size / 2);
        }
    }

    public static double test1(int[] arr1, int[] arr2) {
        // 1.要合并两个数组
        // 要定义一个新数组，然后把两个数组都元素都保存进去
        int[] arr = new int[arr1.length + arr2.length];
        for (int i = 0; i < arr.length; i++) {
            if (i < arr1.length) {
                arr[i] = arr1[i];
            } else {
                arr[i] = arr2[i - arr1.length];
            }
        }

        // 2.对数组排序
        Arrays.sort(arr);
        System.out.println(Arrays.toString(arr));
        // 判断元素个数奇偶
        if (arr.length % 2 == 0) {
            int index = arr.length / 2;
            return (arr[index - 1] + arr[index]) / 2.0;
        } else {
            return arr[arr.length / 2];
        }
    }
}
